#ifndef AUTOAIM_DETECTOR__NUMBER_CLASSIFIER_HPP_
#define AUTOAIM_DETECTOR__NUMBER_CLASSIFIER_HPP_

#include <string>
#include <tuple>

#include <opencv2/dnn.hpp>
#include <opencv2/dnn/all_layers.hpp>
#include <opencv2/opencv.hpp>

namespace detector {
class NumberClassifier {
public:
  using Ptr = std::unique_ptr<NumberClassifier>;

  explicit NumberClassifier(const std::string &model_path);

  std::tuple<int, float> classify(cv::Mat &roi);

private:
  cv::dnn::Net net_;
  bool net_ok_;
};

} // namespace detector
#endif // AUTOAIM_DETECTOR__NUMBER_CLASSIFIER_HPP_